Skip to content

Conversation

@austinroose
Copy link
Collaborator

No description provided.

@Simsuk
Copy link
Collaborator

Simsuk commented Jan 12, 2023

I am not 100% sure but I assumed we are not using Jax. Otherwise we could just copy paste the existing code. The Jax and Pytorch might not be compatible.

@austinroose
Copy link
Collaborator Author

JAX random functions all require an explicit PRNG state to be passed as a first argument, Numpy uses Generator that doesn't need any explicit key to generate random values

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants